from sklearn import neighbors


class WFEstReg:
    def __init__(self):
        self.reg = None

    def fit(self, x, y):
        self.reg = neighbors.KNeighborsRegressor()
        self.reg.fit(x, y)
        return self.reg
